Tail-Aware Adaptive-k: Query-Adaptive Context Selection for Retrieval-Augmented Generation
尾部感知自适应k:查询自适应的RAG上下文选择
🧭 RAG · 自适应上下文选择
AI Lab, 37 Interactive Entertainment · School of Computer Science, Wuhan University
利用排序相似度曲线陡-平-陡模式进行拐点检测,结合局部极值理论检验,实现高效、查询自适应的检索截断,近似oracle质量
📌 工作参考价值(4 点)
- 在RAG pipeline中,可利用相似度曲线的几何特征(陡-平-陡)进行拐点检测,动态决定检索文档数量,避免固定k带来的噪声或遗漏,提升面向电商问答、知识检索的生成质量。
- 推荐系统中,向量召回后的结果截断可借鉴该方法:根据item相似度分布自动确定top-k,替代人工经验调参,降低粗排/精排计算量,尤其在长尾查询下稳定。
- Agent系统需要从工具或知识库中动态选取上下文时,TAA-k提供了一种无训练、低计算开销的自适应截断策略,适合线上实时决策。
- 方法对嵌入模型和压缩维度鲁棒,可跨业务场景复用,无需重新训练或校准。